/*
** Copyright (C) 2010 Calin Radoni
**
** This file is part of CHScanner.
**
** CHScanner is free software: you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation, either version 3 of the License, or
** (at your option) any later version.
**
** CHScanner is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with CHScanner. If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once

#include "MMachineInfo.h"
#include "MGlobalInfo.h"

#include <vector>

class MMachineInfoList
{
private:
	std::vector<MMachineInfo*> theVector;
	std::vector<MMachineInfo*>::iterator iter;
	void CleanUp(void);

	MMachineInfo* CreateMachineInfo(void);
	MMachineInfo* GetMachineInfoByIp(MMachineInfo*);
	MMachineInfo* GetMachineInfoByMac(MMachineInfo*);

	void SortThePorts(void);

public:
	MMachineInfoList(void);
	~MMachineInfoList(void);

	bool Add(MMachineInfo*);
	bool AddIPv6(MMachineInfo*);
	void Clear(void);

	MMachineInfo* GetCreateMachineInfo(DWORD);
	MMachineInfo* GetCreateMachineInfoEx(MFullAddress*, bool*);

	MMachineInfo* GetMachineInfoByMacAddr(MAddressMAC*);
	MMachineInfo* GetMachineInfoByIpAddr(DWORD);
	MMachineInfo* GetMachineInfoByIpAddr6(in_addr6*);

	int GetSize(void);
	MMachineInfo* GetByIdx(int);

	MGlobalInfo globalInfo;

	void SortTheListByIp(void);
	void SortTheListByMAC(void);
	void SortTheListByName(void);
	void RandomizeList(void);
	
	bool SaveToFile(CString fileNameIn);
	bool LoadFromFile(CString fileNameIn);
};
